#ifndef _DENSITY_
#define _DENSITY_

#define K 1
#define N 100

double* peopleDensity;
double* industryDensity;

double getPeopleDensity(double x, double y) {
    return 1;//x*x + y*y + 1;
}

double getIndustryDensity(double x, double y) {
    return 0;//x*x + y*y;
}

void getArray(double (*density)(double, double), double** res) {
    for (int i = 0; i < N; ++i) {
        double iN = double(i) / N;
        for (int j = 0; j < N; ++j) {
            (*res)[i * N + j] = density(iN, double(j) / N);
        }
    }
}

#endif
